// 《围棋》作者版权所有。保留所有权利。
// 此源代码的使用受BSD样式
// 许可证的约束，该许可证可以在许可证文件中找到。

// 包pkgpath确定gccgo/GoLLVM符号使用的包路径。
// 此包不用于gc编译器。
package pkgpath

import (
	"bytes"
	"errors"
	"fmt"
	exec "internal/execabs"
	"io/ioutil"
	"os"
	"strings"
)

// ToSymbolFunc返回一个函数，该函数可用于将
// 包路径转换为适合用作符号的字符串。
// cmd是正在使用的gccgo/GoLLVM编译器，tmpdir是要传递给ioutil的临时
// 目录。临时文件。
// 例如，它返回一个函数，该函数将“net/http”
// 转换为类似“net..z2fhttp”的字符串。
// 不同的gccgo/GoLLVM版本的实际字符串不同，这就是为什么它会返回一个函数
// 来进行适合正在使用的编译器的转换。
func ToSymbolFunc(cmd, tmpdir string) (func(string) string, error) {
	// 为了确定cmd使用的方案，我们编译了一个小的
	// 文件并检查了汇编代码。较旧版本的gccgo 
	// 使用一种简单的损坏方案，其中路径不同但损坏为
	// 相同字符串的包之间可能会发生冲突。最近的版本使用了新的mangler 
	// 以避免这些冲突。
	const filepat = "*_gccgo_manglechck.go"
	f, err := ioutil.TempFile(tmpdir, filepat)
	if err != nil {
		return nil, err
	}
	gofilename := f.Name()
	f.Close()
	defer os.Remove(gofilename)

	if err := ioutil.WriteFile(gofilename, []byte(mangleCheckCode), 0644); err != nil {
		return nil, err
	}

	command := exec.Command(cmd, "-S", "-o", "-", gofilename)
	buf, err := command.Output()
	if err != nil {
		return nil, err
	}

	// 原始破损：开始。卢弗。运行
	// Mangling v2:go。Lu00e4ufer。运行
	// Mangling v3:go_0l_u00e4ufer。Run 
	if bytes.Contains(buf, []byte("go_0l_u00e4ufer.Run")) {
		return toSymbolV3, nil
	} else if bytes.Contains(buf, []byte("go.l..u00e4ufer.Run")) {
		return toSymbolV2, nil
	} else if bytes.Contains(buf, []byte("go.l__ufer.Run")) {
		return toSymbolV1, nil
	} else {
		return nil, errors.New(cmd + ": unrecognized mangling scheme")
	}
}

// mangleCheckCode是我们编译用来确定损坏方案的包。
const mangleCheckCode = `
package läufer
func Run(x int) int {
  return 1
}
`

// toSymbolV1使用原始的损坏方案转换包路径。
func toSymbolV1(ppath string) string {
	clean := func(r rune) rune {
		switch {
		case 'A' <= r && r <= 'Z', 'a' <= r && r <= 'z',
			'0' <= r && r <= '9':
			return r
		}
		return '_'
	}
	return strings.Map(clean, ppath)
}

// toSymbolV2使用第二种损坏方案转换包路径。
func toSymbolV2(ppath string) string {
	// 这必须在bootstrap时间构建，所以它必须使用Go 1.4构建
	// 所以我们不使用字符串。建设者
	bsl := make([]byte, 0, len(ppath))
	changed := false
	for _, c := range ppath {
		if ('A' <= c && c <= 'Z') || ('a' <= c && c <= 'z') || ('0' <= c && c <= '9') || c == '_' {
			bsl = append(bsl, byte(c))
			continue
		}
		var enc string
		switch {
		case c == '.':
			enc = ".x2e"
		case c < 0x80:
			enc = fmt.Sprintf("..z%02x", c)
		case c < 0x10000:
			enc = fmt.Sprintf("..u%04x", c)
		default:
			enc = fmt.Sprintf("..U%08x", c)
		}
		bsl = append(bsl, enc...)
		changed = true
	}
	if !changed {
		return ppath
	}
	return string(bsl)
}

// V3下划线代码从支持下划线编码的字符映射到下划线编码字符。
var v3UnderscoreCodes = map[byte]byte{
	'_': '_',
	'.': '0',
	'/': '1',
	'*': '2',
	',': '3',
	'{': '4',
	'}': '5',
	'[': '6',
	']': '7',
	'(': '8',
	')': '9',
	'"': 'a',
	' ': 'b',
	';': 'c',
}

// toSymbolV3使用第三种损坏方案转换包路径。
func toSymbolV3(ppath string) string {
	// 这必须在boostrap时间构建，所以它必须使用Go 1.4构建
	// 所以我们不使用字符串。建设者
	bsl := make([]byte, 0, len(ppath))
	changed := false
	for _, c := range ppath {
		if ('A' <= c && c <= 'Z') || ('a' <= c && c <= 'z') || ('0' <= c && c <= '9') {
			bsl = append(bsl, byte(c))
			continue
		}

		if c < 0x80 {
			if u, ok := v3UnderscoreCodes[byte(c)]; ok {
				bsl = append(bsl, '_', u)
				changed = true
				continue
			}
		}

		var enc string
		switch {
		case c < 0x80:
			enc = fmt.Sprintf("_x%02x", c)
		case c < 0x10000:
			enc = fmt.Sprintf("_u%04x", c)
		default:
			enc = fmt.Sprintf("_U%08x", c)
		}
		bsl = append(bsl, enc...)
		changed = true
	}
	if !changed {
		return ppath
	}
	return string(bsl)
}
